axis = 3
training_length = 96
predicting_length = 96
forecast_window = 96

splitSize = 0.8
concatScaler = False

GPSType = 1 # 0 one-hot
GPS_stampSize = 1
GPS_size = 31
GPSList = [f'G{str(i).zfill(2)}' for i in range(1, GPS_size + 1)]

loadEpoch = 50
loadModel = f'train_{loadEpoch}.pth'
loadOptim = f'optimizer_{loadEpoch}.pth'
loadSche = f'scheduler_{loadEpoch}.pth'
loadEma = f'ema_{loadEpoch}.pth'
resumeModel = 0
usePTModel = 0 #PreTrained

device = "cuda"
unstd_size = 2 + GPS_stampSize
useHead = True
nll = True
outputSize = 3
outputIdx = 0
stampSize = 7
physicalSize = 5


saveDir = "../../save/"
dataSetDir = "../../dataset/"
toolsDir = "../../tools/"
configModelDir = "../../orbitP/configModel/"
dataFINDir = "../../dataset/dataFIN/"
dataRAPDir = "../../dataset/dataRAP/"
dataSULTDir = "../../dataset/dataSULT/"
dataSULTObs = dataSULTDir + "observation/"
dataSULTPre = dataSULTDir + "prediction/"
dataCPFDir = "../../dataset/dataCPF/"
dataSGP4Dir = "../../dataset/dataSGP4/"
dataOrekitDir = "../../dataset/dataOrekit/"
dataPOEORBDir = "../../dataset/dataPOEORB/"
dataMOEORBDir = "../../dataset/dataMOEORB/"
dataCSV = "../../dataset/dataCSV/"



# saveDir = "./save/"
# dataSetDir = "./dataset/"
# toolsDir = "./tools/"
# configModelDir = "./orbitP/configModel/"
# dataFINDir = "./dataset/dataFIN/"
# dataRAPDir = "./dataset/dataRAP/"
# dataSULTDir = "./dataset/dataSULT/"
# dataSULTObs = dataSULTDir + "observation/"
# dataSULTPre = dataSULTDir + "prediction/"
# dataCPFDir = "./dataset/dataCPF/"
# dataSGP4Dir = "./dataset/dataSGP4/"
# dataOrekitDir = "./dataset/dataOrekit/"
# dataPOEORBDir = "./dataset/dataPOEORB/"
# dataMOEORBDir = "./dataset/dataMOEORB/"
# dataCSV = "./dataset/dataCSV/"

dataObsDir = dataCSV + "dataObs/"
dataPrdDir = dataCSV + "dataPrd/"
dataFinalDir = dataCSV + "dataFinal/"
dataRapidDir = dataCSV + "dataRapid/"
orekitCurPath = toolsDir + "orekit-data-master"
de440Path = toolsDir + "de440.bsp"
omniPath = dataSetDir + "omni.txt"
configPath = saveDir+"save_model/config.txt"
bestConfigPath = saveDir+"save_model/bestConfig.txt"
obsScalerPath = saveDir+"save_model/obsScaler.joblib"
prdScalerPath = saveDir+"save_model/prdScaler.joblib"
lossPath = saveDir+"save_loss/"
predPath = saveDir+"save_predictions/"
modelPath = saveDir+"save_model/"
loadDir = saveDir+"load_model/"
loadPTDir = saveDir+"load_ptmodel/"
showErrorDir = saveDir+'show_error/'
